import torch

DATA_ROOT = "./mnist_project/data"
BATCH_SIZE = 64
LEARNING_RATE = 0.01
EPOCHS = 10
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
MODEL_SAVE_PATH = "mnist_project/model"
